home *** CD-ROM | disk | FTP | other *** search
/ Freelog 100 / FreelogNo100-NovembreDecembre2010.iso / Musique / solfege / solfege-win32-3.17.0.exe / {app} / bin / Lib / bsddb / test / test_pickle.py < prev    next >
Text File  |  2006-04-18  |  2KB  |  76 lines
  1.  
  2. import sys, os, string
  3. import pickle
  4. try:
  5.     import cPickle
  6. except ImportError:
  7.     cPickle = None
  8. import unittest
  9. import glob
  10.  
  11. try:
  12.     # For Pythons w/distutils pybsddb
  13.     from bsddb3 import db
  14. except ImportError, e:
  15.     # For Python 2.3
  16.     from bsddb import db
  17.  
  18.  
  19. #----------------------------------------------------------------------
  20.  
  21. class pickleTestCase(unittest.TestCase):
  22.     """Verify that DBError can be pickled and unpickled"""
  23.     db_home = 'db_home'
  24.     db_name = 'test-dbobj.db'
  25.  
  26.     def setUp(self):
  27.         homeDir = os.path.join(os.path.dirname(sys.argv[0]), 'db_home')
  28.         self.homeDir = homeDir
  29.         try: os.mkdir(homeDir)
  30.         except os.error: pass
  31.  
  32.     def tearDown(self):
  33.         if hasattr(self, 'db'):
  34.             del self.db
  35.         if hasattr(self, 'env'):
  36.             del self.env
  37.         files = glob.glob(os.path.join(self.homeDir, '*'))
  38.         for file in files:
  39.             os.remove(file)
  40.  
  41.     def _base_test_pickle_DBError(self, pickle):
  42.         self.env = db.DBEnv()
  43.         self.env.open(self.homeDir, db.DB_CREATE | db.DB_INIT_MPOOL)
  44.         self.db = db.DB(self.env)
  45.         self.db.open(self.db_name, db.DB_HASH, db.DB_CREATE)
  46.         self.db.put('spam', 'eggs')
  47.         assert self.db['spam'] == 'eggs'
  48.         try:
  49.             self.db.put('spam', 'ham', flags=db.DB_NOOVERWRITE)
  50.         except db.DBError, egg:
  51.             pickledEgg = pickle.dumps(egg)
  52.             #print repr(pickledEgg)
  53.             rottenEgg = pickle.loads(pickledEgg)
  54.             if rottenEgg.args != egg.args or type(rottenEgg) != type(egg):
  55.                 raise Exception, (rottenEgg, '!=', egg)
  56.         else:
  57.             raise Exception, "where's my DBError exception?!?"
  58.  
  59.         self.db.close()
  60.         self.env.close()
  61.  
  62.     def test01_pickle_DBError(self):
  63.         self._base_test_pickle_DBError(pickle=pickle)
  64.  
  65.     if cPickle:
  66.         def test02_cPickle_DBError(self):
  67.             self._base_test_pickle_DBError(pickle=cPickle)
  68.  
  69. #----------------------------------------------------------------------
  70.  
  71. def test_suite():
  72.     return unittest.makeSuite(pickleTestCase)
  73.  
  74. if __name__ == '__main__':
  75.     unittest.main(defaultTest='test_suite')
  76.